package network;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Scanner;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * Created with IntelliJ IDEA.
 * Description:
 * User: Solitudefire
 * Date: 2022-08-09
 * Time: 17:21
 */
public class TcpThreadPoolEchoServer {
    //listen => 英文原译  监听
    //但是在JavaSocket中体现不出来”监听的含义“
    //之所以这么叫，其实是操作系统原生的API里有一个操作叫做Listen
    private ServerSocket serverSocket = null;

    public TcpThreadPoolEchoServer(int port) throws IOException {
        serverSocket = new ServerSocket(port);
    }

    public void start() throws IOException {
        System.out.println("服务器启动！");
        ExecutorService pool = Executors.newCachedThreadPool();
        while (true) {
            //由于TCP是有连接的，不能一上来就读数据，而要先建立连接（接电话）
            //accept()就是在接电话，前提是有人给你打了电话，如果当前没有客户端尝试建立连接，此处的accept就会阻塞。
            //accept返回了一个socket对象，称为clientSocket，后续跟客户端之间的沟通都是通过clientSocket来完成的，
            //在进一步讲，serverSocket就干一件事，接电话
            Socket clientSocket = serverSocket.accept();
            //[改进方法] 在这个地方，每次accept成功，都创建一个新的线程，由新线程负责执行这个processConnection方法
            //可以通过线程池来实现
            pool.submit(new Runnable() {
                @Override
                public void run() {
                    processConnection(clientSocket);
                }
            });
        }
    }

    private void processConnection(Socket clientSocket) {
        System.out.printf("[%s :%d] 客户端建立连接！\n", clientSocket.getInetAddress().toString(), clientSocket.getPort());
        //接下来处理请求和响应
        //这里的针对TCP socket的读写就和文件读写是一模一样的
        try (InputStream inputStream = clientSocket.getInputStream()) {
            try (OutputStream outputStream = clientSocket.getOutputStream()) {
                //循环处理每个请求，分别返回响应
                Scanner scanner = new Scanner(inputStream);
                while (true) {
                    //1.这是读取请求
                    if (!scanner.hasNext()) {
                        System.out.printf("[%s %d] 客户端断开连接\n", clientSocket.getInetAddress().toString(), clientSocket.getPort());
                        break;
                    }
                    //此处用Scanner更方便，如果不用Scanner 就用原生的InputStream 的read也是可以的
                    String request = scanner.next();
                    //2.根据请求计算响应
                    String response = process(request);
                    //3.把这个响应返回给客户端
                    //为了方便起见，可以使用PrintWriter 把OutputStream包裹一下
                    PrintWriter printWriter = new PrintWriter(outputStream);
                    printWriter.println(response);
                    //刷新缓冲区，如果没有这个刷新，可能客户端就不能第一时间看到响应结果，
                    printWriter.flush();

                    System.out.printf("[%s %d] req：%s, resp:%s\n",clientSocket.getInetAddress().toString(),
                            clientSocket.getPort(),request,response);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            //此处要记得关闭操作。
            try {
                clientSocket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    public String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        TcpThreadPoolEchoServer server = new TcpThreadPoolEchoServer(9090);
        server.start();
    }
}
